python3.7 main.py --dataset mnist --input_height=28 --output_height=28 --train